class Constant:
    MODEL_CHOOSE = 2
    MODELS = {
        1: "transformer_encoder",
        2: "LSTM",
        3: "transformer"
    }

    N_INPUT = 6  # 输入的特征维度

    INPUT_WINDOW = 12  # 使用12个月的数据
    OUTPUT_WINDOW = 1  # 预测下1个月的数据

    DEPTH = '100'  # 土壤湿度的深度：'10' '40', '100', '200'
    FEATURES = ['year', DEPTH, 'mon', 'wm2', 'steam_mm', 'raw_mm', 'rain_day']
